package util

import (
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/jinzhu/gorm"
	_ "github.com/joho/godotenv"
)

var _db *gorm.DB
var err error

func init() {
	om := "root:123456@tcp(localhost:3306)/mz_water?charset=utf8"

	_db, err = gorm.Open("mysql", om)
	if err != nil {
		fmt.Printf("mysql connect error %v", err)
	} else {
		fmt.Println("mysql connect success")
	}
	//defer _db.Close()
	// 设置最大空闲连接数
	_db.DB().SetMaxIdleConns(10)
	// 设置最大打开连接数
	_db.DB().SetMaxOpenConns(100)
	_db.LogMode(true)
}

func GetDb() *gorm.DB {
	return _db
}
